from typing import List

from pydantic import BaseModel, Field

from schema.resp_schema.base_schema import BaseRespModel


class LanguageModel(BaseModel):
    code: str
    name: str


class ListLanguageResponseModel(BaseRespModel):
    data: List[LanguageModel] = Field(None)


class LanguageIDResponseModel(BaseModel):
    language_id: int


class AddNewLanguageResponseModel(BaseRespModel):
    data: LanguageIDResponseModel = Field(...)